import java.util.*;

public class Test {
    // 题目1：图像渲染
    // 竟然可以用层序遍历，太惊讶了！
    int[] dx = {0,0,1,-1};
    int[] dy = {1,-1,0,0};

    public int[][] floodFill(int[][] image, int sr, int sc, int color) {
        if(image[sr][sc] == color){
            return image;
        }
        int prev = image[sr][sc];
        int m = image.length;
        int n = image[0].length;
        Queue<int[]> q = new LinkedList<>();
        q.offer(new int[]{sr, sc});
        while(!q.isEmpty()){
            int[] pos = q.poll();
            image[pos[0]][pos[1]] = color;
            int a = pos[0], b = pos[1];
            for(int i = 0; i < 4; i++){
                int x = a + dx[i], y = b + dy[i];
                if(x >= 0 && x < m && y >= 0 && y < n && image[x][y] == prev){
                    q.offer(new int[]{x, y});
                }
            }
        }
        return image;
    }

    // 题目2：岛屿数量
    // 给你一个由 '1'（陆地）和 '0'（水）组成的的二维网格，请你计算网格中岛屿的数量。
    //
    //岛屿总是被水包围，并且每座岛屿只能由水平方向和/或竖直方向上相邻的陆地连接形成。
    //
    //此外，你可以假设该网格的四条边均被水包围。
    // 就是问，有多少个陆地的连通地区

    boolean[][] vis;
    int m, n;
    public int numIslands(char[][] grid) {
        m = grid.length;
        n = grid[0].length;
        vis = new boolean[m][n];
        int ret = 0;
        for(int i = 0; i < m; i++){
            for(int j = 0; j < n; j++){
                if(grid[i][j] == '1' && !vis[i][j]){
                    ret++;
                    BFS(grid, i, j);
                }
            }
        }
        return ret;
    }

    public void BFS(char[][] grid, int i , int j){
        Queue<int[]> q = new LinkedList<>();
        q.offer(new int[]{i, j});
        vis[i][j] = true;
        while(!q.isEmpty()){
            int[] t = q.poll();
            int a = t[0], b = t[1];
            for(int k = 0; k < 4; k++){
                int x = a + dx[k], y = b + dy[k];
                if(x >= 0 && x < m && y >= 0 && y < n && grid[x][y] == '1' && !vis[x][y]){
                    q.offer(new int[]{x, y});
                    vis[x][y] = true;
                }
            }
        }
    }
}
